# -*- coding: utf-8 -*-
# @Author : zhaochen
# @Date   : 2019/3/11
# @File   : R22_IC_NonExternalGuaranteeOvduPeriodsCheck.py
# @E-mail : zhaochen@bbdservice.com


'''eof
name:申请人近5年内贷款、贷记卡和准贷记卡最长逾期期数校验
code:R22_IC_NonExternalGuaranteeOvduPeriodsCheck_2D
tableName:
columnName:
groups:个人征信校验
dependencies:IC_DSJ_2D
type:常用指标
datasourceType:在线指标
description:
eof'''

import sys
import re
import datetime
import pandas as pd
reload(sys)
sys.setdefaultencoding('utf-8')

null_type_list = ['', ' ', None, 'null', 'Null', 'NULL', '/', '-', '--']

def getTable(data, tableName, occurTimes):
    '''
        获取表信息
        :param data:表上层的字典对象
        :param tableName:表的名字
        :param occurTimes:表出现的次数，如果是0代表此表可以不出现，如果是1代表此表必须出现
        :return:该表全部数据
        '''
    table = data.get(tableName)
    if occurTimes == 0:
        if table is None:
            return [{}]
        elif table == "":
            return ""
        elif isinstance(table, list) is False or len(table) == 0:
            return None
        else:
            return data.get(tableName)
    if occurTimes == 1:
        if table == "":
            return ""
        elif table is None or isinstance(table, list) is False or len(table) == 0:
            return None
        else:
            return data.get(tableName)


def isfloat(x):
    '''
    判断x为float类型且大于等于0
    :param x:
    :return:
    '''
    try:
        if isinstance(float(x), float) and float(x)>=0:
            if str(x) in ['inf', 'infinity', 'INF', 'INFINITY', 'True']:
                return False
            else:
                return True
        else:
            return False
    except:
        return False

def checkNull(data,data1, accountType):
    '''若最近24个月还款状态信息为空，返回结果为缺失值，则进行以下逻辑判断：
1.基本信息段（PD01A）-账户类型（PD01AD01）为D1，若最新表现信息段（PD01B）-账户状态（PD01BD01）为3（结清）且关闭日期（PD01BR01）＞5年或账户状态（PD01BD01）为5（转出）且转出月份（PD01BR04）＞5年，则Factor=1,否则Factor=缺失值；
2.基本信息段（PD01A）-账户类型（PD01AD01）为R1,R4，若最新表现信息段（PD01B）-账户状态（PD01BD01）为3（结清）且关闭日期（PD01BR01）＞5年，则Factor=1,否则Factor=缺失值；
3.基本信息段（PD01A）-账户类型（PD01AD01）为R2，R3，若最新表现信息段（PD01B）-账户状态（PD01BD01）为4(销户)且关闭日期（PD01BR01）＞5年或账户状态（PD01BD01）为6(未激活)，则Factor=1,否则Factor=缺失值；
当个人征信数据中账户状态（PD01BD01）为空时，取最近一次月度表现信息段（PD01CD01）;
'''
    if data in null_type_list or data == [{}] or data == 0:
        return 0
    if data1 in null_type_list or data1 == [{}] or data1 == 0:
        PD01CD01 =None
    else:
        PD01CD01 = data1[0].get("PD01CD01")
    # 账户状态
    PD01BD01 = data[0].get("PD01BD01")
    # 关闭日期
    PD01BR01 = data[0].get("PD01BR01")
    PD01BR01 = dateFormat(PD01BR01)
    # 转出月份
    PD01BR04 = data[0].get("PD01BR04")
    PD01BR04 = dateFormat(PD01BR04)
    fiveYearsAgo = (datetime.datetime.now() - pd.tseries.offsets.DateOffset(years=5)).strftime("%Y%m")
    if accountType == "D1":
        if PD01BD01 not in null_type_list:
            if (PD01BD01 == "3" and PD01BR01 is not None and PD01BR01.strftime("%Y%m") <= fiveYearsAgo) \
                    or (PD01BD01 == "5" and PD01BR04 is not None and PD01BR04.strftime("%Y%m") <= fiveYearsAgo):
                return 1
            else:
                return 0
        else:
            if (PD01CD01 is not None and PD01CD01 == "3" and PD01BR01 is not None and PD01BR01.strftime("%Y%m") <= fiveYearsAgo) \
                    or (PD01CD01 is not None and PD01CD01 == "5" and PD01BR04 is not None and PD01BR04.strftime("%Y%m") <= fiveYearsAgo):
                return 1
            else:
                return 0
    elif accountType in ["R1", "R4"]:
        if PD01BD01 not in null_type_list:
            if (PD01BD01 == "3" and PD01BR01 is not None and PD01BR01.strftime("%Y%m") <= fiveYearsAgo):
                return 1
            else:
                return 0
        else:
            if (PD01CD01 is not None and PD01CD01 == "3" and PD01BR01 is not None and PD01BR01.strftime("%Y%m") <= fiveYearsAgo):
                return 1
            else:
                return 0
    elif accountType in ["R2", "R3"]:
        if PD01BD01 not in null_type_list:
            if (PD01BD01 == "4" and PD01BR01 is not None and PD01BR01.strftime("%Y%m") <= fiveYearsAgo) \
                    or (PD01BD01 == "6"):
                return 1
            else:
                return 0
        else:
            if (PD01CD01 is not None and PD01CD01 == "4" and PD01BR01 is not None and PD01BR01.strftime("%Y%m") <= fiveYearsAgo) \
                    or (PD01CD01 == "6"):
                return 1
            else:
                return 0
    else:
        return 0

def dateFormat(time):
    import re
    import datetime
    r = re.compile(r'\D+')
    try:
        date = datetime.datetime.strptime(time, '%Y.%m.%d')
    except:
        try:
            date = datetime.datetime.strptime(time, '%Y-%m-%d')
        except:
            try:
                date = datetime.datetime.strptime(time, '%Y/%m/%d')
            except:
                try:
                    tmp = r.sub('', time)
                    date = datetime.datetime.strptime(tmp, '%Y%m%d')
                except:
                    date = None
    return date


def R22_IC_NonExternalGuaranteeOvduPeriodsCheck():
    data = IC_DSJ_2D['data']
    factor1 = []
    factor2 = []
    # 借贷账户信息
    PDA = getTable(data, 'PDA', 1)
    if PDA is None:
        return u'缺失值'
    elif PDA == "":
        return 0
    else:
        # 借贷账户信息单元
        PD01 = getTable(PDA[0], 'PD01', 0)
        if PD01 is None or PD01 == "":
            return u'缺失值'
        elif PD01 == [{}]:
            return 0
        else:
            for i in PD01:
                try:
                    # 基本信息段
                    PD01A = getTable(i, 'PD01A', 1)
                    # 最近 5 年内历史表现信息段
                    PD01E = getTable(i, 'PD01E', 0)
                    # 最新表现信息段  PD01B  [1..1]
                    PD01B = getTable(i, 'PD01B', 1)
                    # 最近一次月度表现信息段
                    PD01C = getTable(i, 'PD01C', 0)
                    if PD01A is None or PD01E is None or PD01A == "" or PD01E == "":
                        factor1.append(None)
                        factor2.append(None)
                    else:
                        # 账户类型
                        PD01AD01 = PD01A[0].get('PD01AD01')
                        if PD01AD01 in null_type_list:
                            factor1.append(None)
                            factor2.append(None)
                        # D1(非循环贷账户),R1(循环贷账户),R2(贷记卡),R4(循环额度下分账户)
                        elif PD01AD01 in ['D1', 'R1', 'R2', 'R4']:
                            if PD01E == [{}]:
                                factor1.append(0)
                            else:
                                # 还款状态信息
                                PD01EH = getTable(PD01E[0], 'PD01EH', 1)
                                if PD01EH is None or PD01EH == "":
                                    checkRes = checkNull(PD01B,PD01C, PD01AD01)
                                    if checkRes:
                                        factor1.append(0)
                                    else:
                                        factor1.append(None)
                                else:
                                    for PD01EH_Itmes in PD01EH:
                                        # 还款状态
                                        PD01ED01 = PD01EH_Itmes.get('PD01ED01')
                                        if PD01ED01 not in null_type_list:
                                            if isfloat(PD01ED01) and float(PD01ED01) >= 4 and float(PD01ED01) <=7 :
                                                factor1.append(1)
                                            else:
                                                factor1.append(0)
                                        else:
                                            factor1.append(None)
                        # 准贷记卡
                        elif PD01AD01 == 'R3':
                            if PD01E == [{}]:
                                factor2.append(0)
                            else:
                                # 还款状态信息
                                PD01EH = getTable(PD01E[0], 'PD01EH', 1)
                                if PD01EH is None or PD01EH == "":
                                    checkRes = checkNull(PD01B,PD01C, PD01AD01)
                                    if checkRes:
                                        factor2.append(0)
                                    else:
                                        factor2.append(None)
                                else:
                                    for PD01EH_Itmes in PD01EH:
                                        # 还款状态
                                        PD01ED01 = PD01EH_Itmes.get('PD01ED01')
                                        if PD01ED01 not in null_type_list:
                                            if isfloat(PD01ED01) and float(PD01ED01) >= 6 and float(PD01ED01) <= 7:
                                                factor2.append(1)
                                            else:
                                                factor2.append(0)
                                        else:
                                            factor2.append(None)
                        else:
                            factor2.append(0)
                except:
                    factor1.append(None)
                    factor2.append(None)
                    continue

    factor1NotNone = filter(lambda x: x is not None, factor1)
    factor2NotNone = filter(lambda x: x is not None, factor2)
    if len(factor1NotNone) == 0 and len(factor2NotNone) == 0:
        return u'缺失值'
    else:
        return max(sum(factor1NotNone), sum(factor2NotNone))

result = R22_IC_NonExternalGuaranteeOvduPeriodsCheck()
